from api.login_api import LoginAPI


class LoginToken(object):
    @classmethod
    def get_au_headers(cls, body):
        """
        功能：获取认证请求头
        :param body: 登陆的请求体数据
        :return: 成功登陆的认证请求头
        """
        # 发送请求登陆请求
        # 实例化对象，调用方法
        login_api = LoginAPI()
        response = login_api.login(body)
        json_data = response.json()
        print(f"json data = {json_data}")

        # 提取token
        token = json_data.get('data')

        # 设置认证请求头
        headers = {'Authorization': token}

        return headers